package com.atguigu.servlet;

import org.apache.commons.io.IOUtils;
import sun.misc.BASE64Encoder;

import javax.servlet.ServletContext;
import javax.servlet.ServletException;
import javax.servlet.http.HttpServlet;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import java.io.IOException;
import java.io.InputStream;
import java.io.OutputStream;
import java.io.UnsupportedEncodingException;
import java.net.URLEncoder;

public class Download extends HttpServlet{
    public static void main(String[] args) throws UnsupportedEncodingException {
        String content="这是需要Base64编码的内容";
        BASE64Encoder base64Encoder=new BASE64Encoder();
        //执行Base64编码内容
        String encodeString= base64Encoder.encode(content.getBytes("UTF-8"));
        System.out.println(encodeString);
    }
    @Override
    protected void doGet(HttpServletRequest req, HttpServletResponse resp) throws ServletException, IOException {
        //1.获取要下载的文件名
        String downloadFileName="2.jpg";
        //2.读取要下载的文件内容(通过ServletContext对象可以读取)
        ServletContext servletContext = getServletContext();

        //获取要下载的文件类型
        String mimeType = servletContext.getMimeType("/file/" + downloadFileName);
        System.out.println("下载的文件类型"+mimeType);

        //4.在回传前，通过响应头告诉客户端返回数据类型
        resp.setContentType(mimeType);
        //5.还要告诉客户端收到的数据是用于下载使用（还是使用响应头）
        //Content-Disposition响应头，表示收到的数据怎么处理
        //attachment表示附件,表示下载使用
        //filename=表示指定下载的文件名
        //url编码是把汉字转换成%xx%xx的格式

        if(req.getHeader("User-Agent").contains("Firefox")){
            //如果是火狐浏览器使用Base64编码
            resp.setHeader("Content-Disposition","attachment;filename==?UTF-8?B?"+new BASE64Encoder().encode("中国.jpg".getBytes("UTF-8"))+"?=" );
        }else{
            //如果不是火狐，是IE或谷歌，使用URL编码操作
            resp.setHeader("Content-Disposition","attachment;filename"+ URLEncoder.encode("中国.jpg","UTF-8"));
        }
//        resp.setHeader("Content-Disposition","attachment;filename"+downloadFileName);

        /**
         * 斜杠被服务器解析表示地址为http://ip:port/工程名/  映射到代码的web目录
         */
        InputStream resourceAsStream = servletContext.getResourceAsStream("/file/" + downloadFileName);
        //获取响应的输出流
        OutputStream outputStream=resp.getOutputStream();
        //读取输入流中全部的数据，复制给输出流，输出给客户端
        IOUtils.copy(resourceAsStream,outputStream);
        //3.把下载的文件内容回传给客户端


    }
}
